#目标：绘制hr.csv热力图

import seaborn as sns
import matplotlib.pyplot as plt
import pandas as pd

#因为热力图必须将数据转化为浮点数，所以将文件中的部门、薪资列删去
hr = pd.read_csv('../Data/hr.csv', encoding='gbk',usecols=range(0,9))
sns.set_style({'font.sans-serif':['SimHei', 'Arial']})


corr = hr.corr() # 特征的相关系数矩阵
sns.heatmap(data=corr)
plt.title('特征矩阵热力图')
plt.show()

sns.heatmap(corr, annot=True, fmt='.2f')
plt.title('特征矩阵热力图')
plt.show()
